import cv2
import numpy as np
'''ndarray.transpose
用途: cv2读取的图像数据为(H,W,C)形状，不能直接给pytorch使用，
     pytorch需要使用(C,H,W)这种形状的
'''

c=3
h=360
w=640
a=np.ones([c,h,w])
print(a.shape)
b=a.transpose(1, 2, 0) # 意思：维度位置转移，第一维度，第二维度，第0维度
print(b.shape)


